from ultralytics.models import YOLO
import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
 
if __name__ == '__main__':
    model = YOLO(model='ultralytics/cfg/models/11/yolo11.yaml')
    # model.load('yolov8n.pt')
    model.train(data='../datasets/data.yaml', epochs=50, batch=1, device='cpu', imgsz=640, workers=2, cache=False, amp=True, mosaic=False, project='runs/train', name='exp')